import jieba    # 分词

# 设置 Matplotlib 后端（如果需要）
import matplotlib
matplotlib.use('TkAgg')  # 或者 'Agg', 'Qt5Agg', 等，取决于你的系统支持

from matplotlib import pyplot as plt    # 绘图，数据可视化
from wordcloud import WordCloud # 词云
from PIL import Image   # 图片处理
import numpy as np  # 矩阵运算
import sqlite3  # 数据库


# 准备词云所需的句子
conn = sqlite3.connect('movie.db')
c = conn.cursor()
sql = '''select instroduction from movie250'''
data = c.execute(sql)
text = ""
for item in data:
    text += item[0]
    # print(item[0])
print(text)
c.close()
conn.close()

# 分词
cut = jieba.cut(text)
str = ' '.join(cut)
print(len(str))


# 准备
img = Image.open(r'static\assets\img\b.jpg') # 打开遮罩图片
img_array = np.array(img)   # 将图片转换为数据
wc = WordCloud(
    background_color = 'white',
    mask = img_array,
    font_path = r"C:\Windows\Fonts\STXINWEI.TTF",
    scale = 4,  # 缩放
    # width
    # height

)
wc.generate_from_text(str)


# 绘制图片
fig = plt.figure(1)
plt.imshow(wc)
plt.axis('off') # 是否显示坐标轴
# plt.show()  # 显示生成的图片
plt.savefig(r'static\assets\img\b2wc.jpg',dpi=500)  # dpi为清晰度，可选














